# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_pro",
    prompt= [{"role":"user","content":"你是一位文学编辑，请你帮忙改写以下段落，保持原有意思但语言更具文学性，逻辑通顺，语言优美，可适当运用比喻、排比、拟人等修辞。\n在当今社会中，竞争日益激烈，人们需要不断地开拓创新，才能跟上时代的步伐。因此，面对挑战，我们需要勇气和决心。勇气意味着不畏艰险，敢于冒险;决心意味着坚定信念，迎难而上。只有具备这些品质，我们才能在竞争中立于不败之地。"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")